package cn.edu.xust.util;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;

import java.util.Arrays;
import java.util.List;
import java.util.Set;
import java.util.concurrent.TimeUnit;


@Component
public class RedisUtil {

    @Autowired
    private RedisTemplate<String, Object> redisTemplate;
    @Autowired
    private StringRedisTemplate stringRedisTemplate;

    public RedisUtil() {
    }

    /**
     * 设置缓存过期时间
     *
     * @param key
     * @param time
     * @return
     */
    public boolean expire(String key, long time) {
        try {
            if (time > 0) {
                redisTemplate.expire(key, time, TimeUnit.SECONDS);
            }
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }


    /**
     * 根据key 获取过期时间
     *
     * @param key
     * @return 秒
     */
    public long getExpire(String key) {
        return redisTemplate.getExpire(key, TimeUnit.SECONDS);
    }

    /**
     * 判断Key是否存在
     *
     * @param key
     * @return
     */
    public boolean hasKey(String key) {
        try {
            return redisTemplate.hasKey(key);
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    /**
     * 删除键
     *
     * @param key
     */
    public Long del(String... key) {
        if (key != null && key.length > 0) {
            if (key.length == 1) {
                return redisTemplate.delete(key[0])?1L:0L;

            } else {
                return redisTemplate.delete(Arrays.asList(key));

            }
        }
        return 0L;
    }

    /**
     * 获得对象
     *
     * @param key
     * @return
     */
    public Object get(String key) {
        return (key == null) ? null : redisTemplate.boundValueOps(key).get();
    }

    /**
     * 存放缓存
     *
     * @param key
     * @param value
     * @return
     */
    public boolean set(String key, Object value) {
        try {
            redisTemplate.opsForValue().set(key, value);
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    /**
     * 设置缓存 并设置过期时间
     *
     * @param key
     * @param value
     * @param time  TimeUnit.SECONDS
     * @return
     */
    public boolean set(String key, Object value, long time) {
        try {
            if (time > 0) {
                redisTemplate.opsForValue().set(key, value, time, TimeUnit.SECONDS);
            } else {
                set(key, value);
            }
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    /**
     * 递增 key
     *
     * @param key
     * @param delta
     * @return
     */
    public long incr(String key, long delta) {
        if (delta < 0) {
            throw new RuntimeException("递增因子必须大于0");
        }
        return redisTemplate.opsForValue().increment(key, delta);
    }

    /**
     * 递减
     *
     * @param key
     * @param delta
     * @return
     */
    public long decr(String key, long delta) {
        if (delta < 0) {
            throw new RuntimeException("递减因子必须大于0");
        }
        return redisTemplate.opsForValue().increment(key, -delta);
    }


    /**
     * 设置List类型对象专用
     *
     * @param key
     * @param value
     * @param <T>
     */
    public <T> Long setValToListLeft(String key, List<T> value) {
        return this.redisTemplate.opsForList().leftPush(key, value);
    }

    /**
     * 获得List类型对象专用
     *
     * @param key
     * @return
     */
    public Object getValToListLeft(String key) {
        return this.redisTemplate.opsForList().leftPop(key);
    }

    /**
     * 设置List类型对象专用
     *
     * @param key
     * @param value
     * @param <T>
     */
    public <T> Long setValToListRight(String key, List<T> value) {
        return this.redisTemplate.opsForList().rightPush(key, value);
    }

    /**
     * 获得List类型对象专用
     *
     * @param key
     * @return
     */
    public Object getValToListRight(String key) {
        return this.redisTemplate.opsForList().rightPop(key);
    }

    /**
     * 开启Redis 事务
     *
     * @param
     */
    public void multi() {
        // 开启Redis 事务权限
        stringRedisTemplate.setEnableTransactionSupport(true);
        // 开启事务
        stringRedisTemplate.multi();
    }

    /**
     * 提交事务
     *
     * @param
     */
    public void exec() {
        // 成功提交事务
        stringRedisTemplate.exec();
    }

    /**
     * 回滚Redis 事务
     */
    public void discard() {
        // 回滚Redis 事务
        stringRedisTemplate.discard();
    }

    /**
     * 模糊查询keys
     * * 任意多个字符
     * ? 单个字符
     * [] 括号内的某1个字符
     *
     * @return
     */
    public Set<String> getKeys(String condition) {
        return stringRedisTemplate.keys(condition);
    }

    /**
     * 默认锁超时时间
     */
    private final static long DEFAULT_TIMEOUT = 10000;

    public boolean getLock(String key) {
        //value值为系统当前时间
        if (stringRedisTemplate.opsForValue().setIfAbsent(key, String.valueOf(System.currentTimeMillis()))) {
            //设置超时时间 防止死锁
            stringRedisTemplate.expire(key, DEFAULT_TIMEOUT, TimeUnit.MILLISECONDS);
            return true;
        }

        //获取上锁时间 判断锁是否超时，超时则删除key（redis重启会造成重启时间内一些过期的key未被清理）,防止死锁
        String valueTime = stringRedisTemplate.opsForValue().get(key);
        if ((!StringUtils.isEmpty(valueTime)) && System.currentTimeMillis() - Long.parseLong(valueTime) > DEFAULT_TIMEOUT) {
            //释放锁 再次抢锁
            unLock(key);
            return getLock(key);
        }
        return false;


    }

    /**
     * 释放锁
     *
     * @param key
     */
    public void unLock(String key) {
        stringRedisTemplate.opsForValue().getOperations().delete(key);
    }

}
